#source /home/work_nfs11/code/xlgeng/environment/miniconda/etc/profile.d/conda.sh
source ~/.bashrc
#conda activate gxl_base


test_data_dir=/mnt/sfs/asr/test_data/test_sets_format_3000
test_data_dir=/home/work_nfs16/xlgeng/code/wenet_undersdand_and_speech_xlgeng_emotion_only/examples/wenetspeech/whisper/data/test_for_staged_emotion_only

dir=/mnt/sfs/asr/ckpt/qwen2_multi_task_4_6gpus_gxl_adapter/epoch_4_with_speech_gxl_new
ckpt_name=step_24999.pt

dir=/mnt/sfs/asr/ckpt/qwen2_multi_task_4_6gpus_gxl_adapter/epoch_5_with_speech_gxl
ckpt_name=step_21249.pt

dir="/mnt/sfs/asr/code/wenet_undersdand_and_speech_xlgeng/examples/wenetspeech/whisper/exp/epoch_6_with_speech_gxl_new_with_asr-chat"
ckpt_name=step_19999.pt

dir="/mnt/sfs/asr/code/wenet_undersdand_and_speech_xlgeng/examples/wenetspeech/whisper/exp/epoch_6_with_speech_gxl"
ckpt_name="step_76249.pt"

dir="/mnt/sfs/asr/code/wenet_undersdand_and_speech_xlgeng/examples/wenetspeech/whisper/exp/epoch_10_11_with_speech_gxl_with_asr-chat"
ckpt_name="step_51249.pt"
dir="/mnt/sfs/asr/ckpt/qwen2-7B-instruct_multi_task_4_gxl_adapter/epoch_0"
ckpt_name=step_24999.pt
dir=/mnt/sfs/asr/code/wenet_undersdand_and_speech_xlgeng/examples/wenetspeech/whisper/exp/epoch_13_with_speech_gxl_with_asr-chat_full-data
ckpt_name=step_37499.pt
dir=/home/work_nfs16/xlgeng/code/wenet_undersdand_and_speech_xlgeng_emotion_only/examples/wenetspeech/whisper/exp/two_stage_train/stage_1_new
ckpt_name=step_7499.pt
dir=/home/work_nfs15/asr_data/ckpt/understanding_model/only_emotion_from_epoch12_with_ssl_vec/step_28749
ckpt_name=step_28749.pt
dir=/home/work_nfs16/xlgeng/code/wenet_undersdand_and_speech_xlgeng_emotion_only/examples/wenetspeech/whisper/exp/two_stage_train/stage_1_new
ckpt_name=step_24999.pt

data_type="raw" # "shard_full_data"
lang=zh  # en zh
prompt_file=conf/prompt_stage4.yaml


# -------------------format ----------------------------
# test_sets="caption_0107_esc50" # 1W 条
# test_sets="${test_sets// /---}"
# gpu_id=0
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="caption_0107_vocalsound  public_test/roobo_100"
# test_sets="${test_sets// /---}"
# gpu_id=1
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="public_test/kaggle_gender public_test/kaggle_age"
# test_sets="${test_sets// /---}"
# gpu_id=2
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="public_test/aishell1_gender"
# test_sets="${test_sets// /---}"
# gpu_id=3
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file


# test_sets="age aslp_chat_test public_test/AirBench_speech chat"
# test_sets="${test_sets// /---}"
# gpu_id=4
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="style"
# test_sets="${test_sets// /---}"
# gpu_id=5
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="caption_aslp_record emotion"
# test_sets="${test_sets// /---}"
# gpu_id=6
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="caption_aslp_record gender public_test/MER23_test public_test/MELD_test"
# test_sets="${test_sets// /---}"
# gpu_id=7
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="aligin_english_clean"
# test_sets="${test_sets// /---}"
# gpu_id=5
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file


# test_sets="align_cn_noize"
# test_sets="${test_sets// /---}"
# gpu_id=6
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file


# test_sets="align_en_noize"
# test_sets="${test_sets// /---}"
# gpu_id=7
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file


# # ----------------format --------------------------------


# ----------------format for staged emotion only --------------------------------
# test_sets="stage_1 stage_2"
# test_data_dir=/home/work_nfs16/xlgeng/code/wenet_undersdand_and_speech_xlgeng_emotion_only/examples/wenetspeech/whisper/data/test_for_staged_emotion_only
# test_sets="${test_sets// /---}"
# gpu_id=4
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# dir=/home/work_nfs16/xlgeng/code/wenet_undersdand_and_speech_xlgeng_emotion_only/examples/wenetspeech/whisper/exp/two_stage_train/stage_1_new
# ckpt_name=step_24999.pt
# test_data_dir=/home/work_nfs16/zxzhao/workspace/SSL_LLM/LLM/stage1

# test_sets="casia"
# test_sets="${test_sets// /---}"
# gpu_id=1
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="meld"
# test_sets="${test_sets// /---}"
# gpu_id=2
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="esd_test"
# test_sets="${test_sets// /---}"
# gpu_id=3
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file






# dir=/home/work_nfs16/xlgeng/code/wenet_undersdand_and_speech_xlgeng_emotion_only/examples/wenetspeech/whisper/exp/two_stage_train/stage_2_plus_meld
# ckpt_name=step_9999.pt
# test_data_dir=/home/work_nfs16/zxzhao/workspace/SSL_LLM/LLM/stage2

# test_sets="asvpesd casia"
# test_sets="${test_sets// /---}"
# gpu_id=4
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="EMO-Emilia emovdb"
# test_sets="${test_sets// /---}"
# gpu_id=5
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="esd_test"
# test_sets="${test_sets// /---}"
# gpu_id=6
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# test_sets="m3ed meld mesd"
# test_sets="${test_sets// /---}"
# gpu_id=7
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file



# ----------------format for staged emotion only --------------------------------

# dir=/home/work_nfs16/xlgeng/code/wenet_undersdand_and_speech_xlgeng_emotion_only/examples/wenetspeech/whisper/exp/two_stage_train/only_ssl_from_zero
# ckpt_name=step_19999.pt
# test_data_dir=/home/work_nfs16/zxzhao/workspace/SSL_LLM/LLM/stage2
# test_sets="EMO-Emilia"
# test_sets="${test_sets// /---}"
# gpu_id=0
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file

# 原生的um ckpt
# dir=/home/work_nfs15/asr_data/ckpt/understanding_model/epoch12/step_51249
# ckpt_name=step_51249.pt
# test_data_dir=/home/work_nfs16/zxzhao/workspace/SSL_LLM/LLM/stage1
# test_sets="EMO-Emilia"
# test_sets="${test_sets// /---}"
# gpu_id=4
# my_named_nohup special_${gpu_id}_${ckpt_name}_next bash gxl_decode/decode_common_gxl_zhiding_prompt.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file


dir=/home/work_nfs15/asr_data/ckpt/understanding_model/only_emotion_from_epoch12_with_ssl_vec/step_28749
ckpt_name=step_28749.pt
test_data_dir=/home/xlgeng/tmp
test_sets="EMO-Emilia"
test_sets="${test_sets// /---}"
gpu_id=4
my_named_nohup special_${gpu_id}_${ckpt_name}_next_node54 bash gxl_decode/decode_common_gxl_zhiding_prompt.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file



# ----------------format for staged emotion only 22222222 --------------------------------
# test_data_dir=/home/work_nfs15/asr_data/data/test_sets_format_3000
# test_sets="emotion public_test/MELD_test public_test/MER23_test"
# test_sets="emotion"
# test_sets="${test_sets// /---}"
# gpu_id=4
# my_named_nohup special_${gpu_id}_${ckpt_name} bash gxl_decode/decode_common_gxl.sh  --data_type $data_type --test_sets "$test_sets" --test_data_dir $test_data_dir --gpu_id $gpu_id --dir $dir --ckpt_name $ckpt_name --lang $lang --prompt_file $prompt_file
# ----------------format for staged emotion only --------------------------------
